package com.lkd.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.google.common.base.Strings;
import com.google.common.collect.Lists;
import com.lkd.common.VMSystem;
import com.lkd.config.TopicConfig;
import com.lkd.contract.SupplyChannel;
import com.lkd.contract.SupplyContract;
import com.lkd.contract.TaskCompleteContract;
import com.lkd.dao.TaskDao;
import com.lkd.emq.MqttProducer;
import com.lkd.entity.TaskDetailsEntity;
import com.lkd.entity.TaskEntity;
import com.lkd.entity.TaskStatusTypeEntity;
import com.lkd.exception.LogicException;
import com.lkd.feign.UserService;
import com.lkd.feign.VMService;
import com.lkd.http.vo.CancelTaskReq;
import com.lkd.http.vo.TaskCollectVO;
import com.lkd.http.vo.TaskReq;
import com.lkd.service.TaskDetailsService;
import com.lkd.service.TaskService;
import com.lkd.service.TaskStatusTypeService;
import com.lkd.vo.*;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.util.ObjectUtils;
import org.springframework.util.StringUtils;

import java.time.Duration;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.Month;
import java.time.format.DateTimeFormatter;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

@Service
@Slf4j
public class TaskServiceImpl extends ServiceImpl<TaskDao,TaskEntity> implements TaskService{

    @Autowired
    private TaskStatusTypeService statusTypeService;


    @Override
    public Pager<TaskEntity> search(Long pageIndex, Long pageSize, String innerCode, Integer userId, String taskCode, Integer status, Boolean isRepair, String start, String end) {
        Page<TaskEntity> page = new Page<>(pageIndex,pageSize);
        LambdaQueryWrapper<TaskEntity> qw = new LambdaQueryWrapper<>();
        if(!Strings.isNullOrEmpty(innerCode)){
            qw.eq(TaskEntity::getInnerCode,innerCode);
        }
        if(userId != null && userId > 0){
            qw.eq(TaskEntity::getUserId,userId);
        }
        if(!Strings.isNullOrEmpty(taskCode)){
            qw.like(TaskEntity::getTaskCode,taskCode);
        }
        if(status != null && status > 0){
            qw.eq(TaskEntity::getTaskStatus,status);
        }
        if(isRepair != null){
            if(isRepair){
                qw.ne(TaskEntity::getProductTypeId, VMSystem.TASK_TYPE_SUPPLY);
            }else {
                qw.eq(TaskEntity::getProductTypeId,VMSystem.TASK_TYPE_SUPPLY);
            }
        }
        if(!Strings.isNullOrEmpty(start) && !Strings.isNullOrEmpty(end)){
            qw
                    .ge(TaskEntity::getCreateTime, LocalDate.parse(start, DateTimeFormatter.ISO_LOCAL_DATE))
                    .le(TaskEntity::getCreateTime,LocalDate.parse(end,DateTimeFormatter.ISO_LOCAL_DATE));
        }
        //根据最后更新时间倒序排序
        qw.orderByDesc(TaskEntity::getUpdateTime);

        return Pager.build(this.page(page,qw));
    }



    @Override
    public List<TaskStatusTypeEntity> getAllStatus() {
        QueryWrapper<TaskStatusTypeEntity> qw = new QueryWrapper<>();
        qw.lambda()
                .ge(TaskStatusTypeEntity::getStatusId,VMSystem.TASK_STATUS_CREATE);

        return statusTypeService.list(qw);
    }

    @Autowired
    private RedisTemplate<String,Object> redisTemplate;

    @Autowired
    private TaskDetailsService taskDetailsService;

    @Autowired
    private VMService vmService;

    @Autowired
    private UserService userService;

    @Autowired
    private MqttProducer mqttProducer;


    /*
    * 业务分析：
    *   0.是否开启事务
    *   1.判断关键数据
    *   2.判断业务数据
    *       售货机-售货机编号来查询数据（Feign）
    *           判断售货机是否存在
    *           判断售货机的状态
    *               补货工单-> 只能运营
    *               投放工单-> 未投放
    *               撤机工单 -> 只能运营
    *           判断同一个售货机的同种工单是否有未完成:工单表 select count(*) from tb_task where inner_code =? and production_type = ? and task_status <= 2
    *       人员（运维和运营）
    *           判断是否存在
    *           所在区域和售后机是否一致
    *       工单
    *           如果是补货工单，那么补货工单详细的集合不能为空
    *    3.业务操作
    *       保存工单数据
    *           前端传递过来的值进行赋值到Entity
    *           生成工单编号、工单状态
    *       如果是补货工单
    *           保存工单详情数据（将整个的结合数据保存-for）
    *    4.返回结果内容
    *
    * */
    @Transactional
    @Override
    public boolean createTask(TaskReq taskReq) throws LogicException {

        //1.判断关键数据
        verifyTaskMsg(taskReq);


        // 2.判断业务数据
        //     2.1 售货机是否存在
        //         不存在抛出异常
        //跨服务查询售货机微服务，得到售货机的地址和区域id
        VmVO vm = vmService.getVMInfo(taskReq.getInnerCode());
        if(vm==null){
            throw new LogicException("该机器不存在");
        }

        //     2.2 售货机的状态
        //         如果是投放工单，状态为运营则抛出异常
        //         如果是补货工单，状态不是运营状态则抛出异常
        //         如果是撤机工单，状态不是运营状态则抛出异常
        checkCreateTask(vm.getVmStatus(),taskReq.getProductType());



        //     2.3 同一个售货机下的同种工单是否完成
        //         同一个售货机下的同种工单是能有一个
        if(hasTask(taskReq.getInnerCode(),taskReq.getProductType())) {
            throw new LogicException("该机器有未完成的同类型工单");
        }


        //     2.4 判断指派用户数据是否存在
        //         不存在需要抛出异常
        //跨服务查询用户微服务，得到用户名
        UserVO user = userService.getUser(taskReq.getUserId());
        if(user==null){
            throw new LogicException("该用户不存在");
        }


        if (!(ObjectUtils.nullSafeEquals(user.getRegionId(),vm.getRegionId()))) {
            throw new LogicException("非同区域下的人员不能接受工单");
        }

        // 2.5 如果是补货工单，那么补货工单详细的集合不能为空
        if (VMSystem.TASK_TYPE_SUPPLY == taskReq.getProductType()) {
            if (CollectionUtils.isEmpty(taskReq.getDetails())) {
                throw new LogicException("补货工单的详情数据不能为空");
            }
        }


        // 3.添加售货机工单信息
        //     3.1 保存工单信息
        //新增工单表记录
        TaskEntity taskEntity = new TaskEntity();
        taskEntity.setTaskCode(generateTaskCode());//工单编号
        BeanUtils.copyProperties(taskReq,taskEntity);//复制属性
        taskEntity.setTaskStatus(VMSystem.TASK_STATUS_CREATE);//工单状态
        taskEntity.setProductTypeId(taskReq.getProductType());//工单类型

        taskEntity.setAddr(vm.getNodeAddr());//地址
        taskEntity.setRegionId(  vm.getRegionId() );//区域
        taskEntity.setUserName(user.getUserName());//用户名
        this.save(taskEntity);


        //     3.2 如果是补货工单，需要添加补货工单详情数据
        //         补货工单详情数据不能为空
        //如果是补货工单，向 工单明细表插入记录
        if(taskEntity.getProductTypeId() == VMSystem.TASK_TYPE_SUPPLY){
            taskReq.getDetails().forEach(d->{
                TaskDetailsEntity detailsEntity = new TaskDetailsEntity();
                BeanUtils.copyProperties( d,detailsEntity );
                detailsEntity.setTaskId(taskEntity.getTaskId());
                taskDetailsService.save(detailsEntity);
            });
        }

        updateTaskZSet(taskEntity,1);

        return true;
    }

    /*判断工单基础数据*/
    private void verifyTaskMsg(TaskReq taskReq) {
        if (StringUtils.isEmpty(taskReq.getInnerCode())) {
            throw new LogicException("售货机编号不能为空");
        }

        if (ObjectUtils.isEmpty(taskReq.getUserId())) {
            throw new LogicException("必须要指定员工");

        }
        if (ObjectUtils.isEmpty(taskReq.getProductType())) {
            throw new LogicException("工单类型不能为空");
        }

        if (StringUtils.isEmpty(taskReq.getDesc())) {
            throw new LogicException("工单描述不能为空");
        }

        if (VMSystem.TASK_TYPE_SUPPLY == taskReq.getProductType()) {
            if (CollectionUtils.isEmpty(taskReq.getDetails())) {
                throw new LogicException("补货工单补货详情不能为空");
            }
        }

    }

    /**
     * 生成工单编号
     * @return
     */
    private String generateTaskCode(){
        //日期+序号
        String date = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));  //日期字符串
        String key= "lkd.task.code."+date; //redis key
        Object obj = redisTemplate.opsForValue().get(key);
        if(obj==null){
            redisTemplate.opsForValue().set(key,1L, Duration.ofDays(1) );
            return date+"0001";
        }
        return date+  Strings.padStart( redisTemplate.opsForValue().increment(key,1).toString(),4,'0');
    }


    /**
     * 创建工单校验
     * @param vmStatus
     * @param productType
     * @throws LogicException
     */
    private void checkCreateTask(Integer vmStatus,int productType) throws LogicException {
        //如果是投放工单，状态为运营
        if(productType == VMSystem.TASK_TYPE_DEPLOY  && vmStatus == VMSystem.VM_STATUS_RUNNING){
            throw new LogicException("该设备已在运营");
        }

        //如果是补货工单，状态不是运营状态
        if(productType == VMSystem.TASK_TYPE_SUPPLY  && vmStatus != VMSystem.VM_STATUS_RUNNING){
            throw new LogicException("该设备不在运营状态");
        }

        //如果是撤机工单，状态不是运营状态
        if(productType == VMSystem.TASK_TYPE_REVOKE  && vmStatus != VMSystem.VM_STATUS_RUNNING){
            throw new LogicException("该设备不在运营状态");
        }
    }


    /**
     * 同一台设备下是否存在未完成的工单
     * @param innerCode
     * @param productionType
     * @return
     */
    private boolean hasTask(String innerCode,int productionType){
        QueryWrapper<TaskEntity> qw = new QueryWrapper<>();
        qw.lambda()
                .select(TaskEntity::getTaskId)
                .eq(TaskEntity::getInnerCode,innerCode)
                .eq(TaskEntity::getProductTypeId,productionType)
                .le(TaskEntity::getTaskStatus,VMSystem.TASK_STATUS_PROGRESS);
        return this.count(qw) > 0;
    }


    /*
    * 业务分析：
    *   1.判断关键数据
    *   2.判断业务数据
    *       工单
    *           是否存在
    *           工单的状态
    *           接受工单的人是否有权
    *   3.业务操作
    *       修改工单的状态为：已接收
    * */

    /*
     * 业务分析：
     *   1.判断关键数据
     *   2.判断业务数据
     *       工单数据
     *         判断订单是否存在
     *         判断其状态：如果不是 创建(待处理)则抛出异常
     *         判断运维或运营是否有权接受工单
     *   3.修改订单的状态
     * */
    @Override
    public boolean accept(Long id,Integer userId) {
        // 1.判断关键数据
        if (ObjectUtils.isEmpty(id)) {
            throw new LogicException("工单的关键数据不能为空");
        }


        // 2.判断业务数据
        //     工单数据
        //       判断订单是否存在
        TaskEntity task = this.getById(id);  //查询工单
        if (ObjectUtils.isEmpty(task)) {
            throw new LogicException("工单数据不存在");
        }
        //       判断其状态：如果不是 创建(待处理)则抛出异常
        if(task.getTaskStatus()!= VMSystem.TASK_STATUS_CREATE ){
            throw new LogicException("工单状态不是待处理");
        }
        //       判断运维或运营是否有权接受工单
        if(!(ObjectUtils.nullSafeEquals(userId,task.getUserId())) ){
            throw new LogicException("操作非法");
        }

        // 3.修改订单的状态
        // update tb_task set task_status = ? where taskid = ?
        var updateWrapper = new LambdaUpdateWrapper<TaskEntity>();
        updateWrapper.set(TaskEntity::getTaskStatus, VMSystem.TASK_STATUS_PROGRESS);
        updateWrapper.set(TaskEntity::getUpdateTime, LocalDateTime.now());
        updateWrapper.eq(TaskEntity::getTaskId, task.getTaskId());
        return this.update(updateWrapper);
    }


    /*
     * 业务分析：
     *   1.判断关键数据
     *   2.判断业务数据
     *       工单数据
     *         判断订单是否存在
     *         判断其状态：如果是 完成 或 取消  则抛出异常
     *         判断运维或运营是否有权接受工单
     *   3.修改订单的状态
     * */
    @Override
    public boolean cancelTask(Long id,Integer userId, CancelTaskReq cancelTaskReq) {
        // 2.判断业务数据
        //     工单数据
        //       判断订单是否存在
        TaskEntity task = this.getById(id);  //查询工单
        if (ObjectUtils.isEmpty(task)) {
            throw new LogicException("工单数据不存在");
        }
        //       判断其状态：如果是 完成 或 取消  则抛出异常
        if(task.getTaskStatus()== VMSystem.TASK_STATUS_FINISH  || task.getTaskStatus()== VMSystem.TASK_STATUS_CANCEL ){
            throw new LogicException("工单状态异常");
        }
        //       判断运维或运营是否有权接受工单
        if(!(ObjectUtils.nullSafeEquals(userId,task.getUserId())) ){
            throw new LogicException("操作非法");
        }


        // 3.修改订单的状态
        task.setTaskStatus( VMSystem.TASK_STATUS_CANCEL  );
        task.setDesc(cancelTaskReq.getDesc());
        return this.updateById(task);
    }

    /*
     * 业务分析：
     *   1.判断关键数据（此处使用Spring Validataion）
     *   2.判断业务数据
     *       工单数据
     *         判断订单是否存在
     *         判断其状态：如果是 完成 或 取消  则抛出异常
     *         判断运维或运营是否有权接受工单
     *   3.修改订单的状态
     *   4.修改售货机状态（后面使用mq实现-暂且不实现）
     *       如果是投放工单或撤机工单：更新售货机状态
     *       如果是补货工单：更新售货机库存
     * */
    @Override
    public boolean completeTask(long id,Integer userId) {

        // 1.判断关键数据（此处使用Spring Validataion）

        // 2.判断业务数据
        //     工单数据
        //       判断订单是否存在
        TaskEntity taskEntity = this.getById(id);  //查询工单
        if (ObjectUtils.isEmpty(taskEntity)) {
            throw new LogicException("工单数据不存在");
        }


        //如果是投放工单或撤机工单
        if(taskEntity.getProductTypeId()==VMSystem.TASK_TYPE_DEPLOY
                || taskEntity.getProductTypeId()==VMSystem.TASK_TYPE_REVOKE){
            noticeVMServiceStatus(taskEntity);//运维工单封装与下发
        }

        //       判断运维或运营是否有权接受工单
        if(!(ObjectUtils.nullSafeEquals(userId,taskEntity.getUserId())) ){
            throw new LogicException("操作非法");
        }

        // 3.修改订单的状态
        taskEntity.setTaskStatus(VMSystem.TASK_STATUS_FINISH);//修改工单状态
        this.updateById(taskEntity);


        // 4.修改售货机状态
        //如果是投放工单或撤机工单
        if(taskEntity.getProductTypeId()==VMSystem.TASK_TYPE_DEPLOY
                || taskEntity.getProductTypeId()==VMSystem.TASK_TYPE_REVOKE){
            noticeVMServiceStatus(taskEntity);//运维工单封装与下发
        }
        //如果是补货工单!
        if(taskEntity.getProductTypeId()==VMSystem.TASK_TYPE_SUPPLY){
            noticeVMServiceSupply(taskEntity);
        }
        return true;
    }




    /**
     * 运维工单封装与下发
     * @param taskEntity
     */
    private void noticeVMServiceStatus(TaskEntity taskEntity){
        //向消息队列发送消息，通知售货机更改状态
        //封装协议
        TaskCompleteContract taskCompleteContract=new TaskCompleteContract();
        taskCompleteContract.setInnerCode(taskEntity.getInnerCode());//售货机编号
        taskCompleteContract.setTaskType( taskEntity.getProductTypeId() );//工单类型
        //发送到emq
        try {
            mqttProducer.send( TopicConfig.VMS_COMPLETED_TOPIC,2, taskCompleteContract );
        } catch (Exception e) {
            log.error("发送工单完成协议出错");
            throw new LogicException("发送工单完成协议出错");
        }
    }


    /**
     * 补货协议封装与下发
     * @param taskEntity
     */
    private void noticeVMServiceSupply(TaskEntity taskEntity){

        //协议内容封装
        //1.根据工单id查询工单明细表
        QueryWrapper<TaskDetailsEntity> qw = new QueryWrapper<>();
        qw.lambda()
                .eq(TaskDetailsEntity::getTaskId,taskEntity.getTaskId());
        List<TaskDetailsEntity> details = taskDetailsService.list(qw);
        //2.构建协议内容
        SupplyContract supplyContract = new SupplyContract();
        supplyContract.setInnerCode(taskEntity.getInnerCode());//售货机编号
        List<SupplyChannel> supplyChannels = Lists.newArrayList();//补货数据
        //从工单明细表提取数据加到补货数据中
        details.forEach(d->{
            SupplyChannel channel = new SupplyChannel();
            channel.setChannelId(d.getChannelCode());//货道编号
            channel.setCapacity(d.getExpectCapacity());//补货数量
            supplyChannels.add(channel);
        });
        supplyContract.setSupplyData(supplyChannels);

        //3.下发补货协议
        //发送到emq
        try {
            mqttProducer.send( TopicConfig.VMS_SUPPLY_TOPIC,2, supplyContract );
        } catch (Exception e) {
            log.error("发送工单完成协议出错");
            throw new LogicException("发送工单完成协议出错");
        }
    }


    /**
     * 更新工单量列表
     * @param taskEntity
     * @param score 增加的分值
     */
    private void  updateTaskZSet(TaskEntity taskEntity,int score){
        // 1.根据工单的类型来设置用户角色code
        var roleCode=VMSystem.ROLE_OPERATIONS_CODE;//运维员
        if(taskEntity.getProductTypeId().intValue()==VMSystem.TASK_TYPE_SUPPLY){//如果是补货工单
            roleCode=VMSystem.ROLE_OPERATING_CODE;//运营员
        }

        // 2.构建存储的key并更改zset的score值
        String key= VMSystem.REGION_TASK_KEY_PREF
                + LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"))
                +"."+ taskEntity.getRegionId()+"."+roleCode;
        redisTemplate.opsForZSet().incrementScore(key,taskEntity.getUserId(),score);
    }



    @Override
    public int getLeastUser(Long regionId, Boolean isRepair) {

        // 1.根据工单的类型来设置用户角色code
        var roleCode=VMSystem.ROLE_OPERATING_CODE;
        if(isRepair){ //如果是运维工单
            roleCode=VMSystem.ROLE_OPERATIONS_CODE;
        }

        // 2.构建存储的key并从zset中获得当日工单最少的员工
        String key= VMSystem.REGION_TASK_KEY_PREF
                + LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"))
                +"."+ regionId+"."+roleCode;
        Set<Object> set = redisTemplate.opsForZSet().range(key, 0, 0);
        if(set==null || set.isEmpty()){
           return 0;  //如果没有数据则返回0
        }

        return (Integer)set.stream().findFirst().get();
    }


    @Override
    public List<TaskReportInfoVO> getTaskReportInfo(LocalDateTime start, LocalDateTime end) {

        //运营工单总数total
        var supplyTotal = this.taskCount( start,end,false, null );
        //运维工单总数
        var repairTotal = this.taskCount( start,end,true, null );


        //完成的运营工单总数
        var completedSupply = this.taskCount( start,end,false, VMSystem.TASK_STATUS_FINISH );
        //完成的运维工单总数
        var completedRepair = this.taskCount( start,end,true, VMSystem.TASK_STATUS_FINISH );


        //拒绝掉的运营工单总数
        var cancelSupply = this.taskCount( start,end,false, VMSystem.TASK_STATUS_CANCEL );
        //拒绝掉的运维工单总数
        var cancelRepair = this.taskCount( start,end,true, VMSystem.TASK_STATUS_CANCEL );


        // 获取运营人员数量
        var operatorCount = userService.getOperatorCount();
        //获取运维人员总数
        var repairerCount = userService.getRepairerCount();



        List<TaskReportInfoVO> result = Lists.newArrayList();

        //运营人员
        var supplyTaskInfo = new TaskReportInfoVO();
        supplyTaskInfo.setTotal(supplyTotal);
        supplyTaskInfo.setCancelTotal(cancelSupply);
        supplyTaskInfo.setCompletedTotal(completedSupply);
        supplyTaskInfo.setRepair(false);
        supplyTaskInfo.setWorkerCount(operatorCount);
        result.add(supplyTaskInfo);

        //运维人员
        var repairTaskInfo = new TaskReportInfoVO();
        repairTaskInfo.setTotal(repairTotal);
        repairTaskInfo.setCancelTotal(cancelRepair);
        repairTaskInfo.setCompletedTotal(completedRepair);
        repairTaskInfo.setRepair(true);
        repairTaskInfo.setWorkerCount(repairerCount);
        result.add(repairTaskInfo);
        return result;
    }




    /**
     * 统计工单数量
     * @param start
     * @param end
     * @param repair 是否是运维工单
     * @param taskStatus
     * @return
     */
    private int taskCount(LocalDateTime start, LocalDateTime end , Boolean repair , Integer taskStatus ){
        // select count(*) from tb_task where update_time >= ? and update_time <= ? and   (总)

        // select count(*) from tb_task where update_time >= ? and update_time <= ? and
        //                                    task_status（工单状态） = ?  and product_type_id(工单类型) = ?

        LambdaQueryWrapper<TaskEntity> qw = new LambdaQueryWrapper<>();
        qw.ge(TaskEntity::getUpdateTime,start)
                .le(TaskEntity::getUpdateTime,end);


        //按工单状态查询
        if(taskStatus!=null){
            qw.eq(TaskEntity::getTaskStatus,taskStatus);
        }
        if(repair){//如果是运维工单
            qw.ne(TaskEntity::getProductTypeId,VMSystem.TASK_TYPE_SUPPLY);
        }else{
            qw.eq(TaskEntity::getProductTypeId,VMSystem.TASK_TYPE_SUPPLY);
        }
        return this.count(qw);
    }


    @Override
    public List<UserWorkVO> getUserWorkTop10(LocalDate startTime, LocalDate endTime, Boolean isRepair, Long regionId) {
        // select
        //     user_name,
        //     count(*) user_id
        // from
        //     tb_task
        // where
        //     task_status = 4
        //     and
        //     update_time >= '2022-12-01 00:00:00'
        //     and
        //     update_time < '2023-01-01 00:00:00'
        //     and
        //     region_id = ?
        //     and
        //     product_type_id = ?
        // group by user_name
        // order by user_id desc
        // limit 10
        LambdaQueryWrapper<TaskEntity> taskEntityLambdaQueryWrapper = new LambdaQueryWrapper<>();
        taskEntityLambdaQueryWrapper.select(TaskEntity::getTaskId);


        var queryWrapper = new QueryWrapper<TaskEntity>();
        queryWrapper.select(" user_name,count(*) user_id ");
        queryWrapper.last("order by user_id desc  limit 10");
        queryWrapper.lambda()
                .eq(TaskEntity::getTaskStatus, VMSystem.TASK_STATUS_FINISH)
                .ge(TaskEntity::getUpdateTime, startTime)
                .lt(TaskEntity::getUpdateTime, endTime.plusDays(1))
                .groupBy(TaskEntity::getUserName);

        if(regionId >0){
            queryWrapper.lambda().eq(TaskEntity::getRegionId,regionId);
        }
        if(isRepair){
            queryWrapper.lambda().ne(TaskEntity::getProductTypeId,VMSystem.TASK_TYPE_SUPPLY);
        }else {
            queryWrapper.lambda().eq(TaskEntity::getProductTypeId,VMSystem.TASK_TYPE_SUPPLY);
        }

        var resultList = this.list(queryWrapper);


        // 将 List<TaskEntity> 变为 List<UserWorkVO>
        var userWorks = resultList.stream().map(taskEntity -> {
            var userWorkVO = new UserWorkVO();
            userWorkVO.setUserName(taskEntity.getUserName());
            userWorkVO.setWorkCount(taskEntity.getUserId());
            return userWorkVO;
        }).collect(Collectors.toList());

        return userWorks;
    }



    @Override
    public List<TaskCollectVO> getTaskReport(LocalDate startTime, LocalDate endTime) {
        List<TaskCollectVO> taskCollectVOList=Lists.newArrayList();
        //从开始日期到截至日期，逐条统计
        startTime.datesUntil(endTime.plusDays(1))
                .forEach(date->{
                    var taskCollectVO=new TaskCollectVO();
                    taskCollectVO.setCollectDate(date);
                    taskCollectVO.setProgressCount(count(date,VMSystem.TASK_STATUS_PROGRESS )); //进行中工单数
                    taskCollectVO.setFinishCount(count(date,VMSystem.TASK_STATUS_FINISH ));//完成工单数
                    taskCollectVO.setCancelCount(count(date,VMSystem.TASK_STATUS_CANCEL ));//取消工单数
                    taskCollectVOList.add(taskCollectVO);
                });
        return taskCollectVOList;
    }


    /*  测试日期 datesUntil 方法 */
    public static void main(String[] args) {
        var startTime = LocalDate.of(2022, Month.NOVEMBER, 1); //2022-11-01
        var endTime = LocalDate.of(2022, Month.NOVEMBER, 12); //2022-11-12

        startTime.datesUntil(endTime.plusDays(1)).forEach(date->{
            System.out.println(date);
        });
    }


    /**
     * 按时间和状态进行统计
     * @param start
     * @param taskStatus
     * @return
     */
    private int count(LocalDate start ,Integer taskStatus){

        /*
        *   select
                count(*)
            from
                tb_task
            where
                update_time >= '2022-11-01 00:00:00'
                and
                update_time < '2022-11-02 00:00:00'
                and
                task_status = ?
        * */
        var qw = new LambdaQueryWrapper<TaskEntity>();
        qw
                .ge(TaskEntity::getUpdateTime,start)
                .lt(TaskEntity::getUpdateTime,start.plusDays(1))
                .eq(TaskEntity::getTaskStatus, taskStatus);
        return this.count(qw);
    }


    @Override
    public UserWorkVO getUserWork(Integer userId, LocalDateTime start, LocalDateTime end) {
        var userWork = new UserWorkVO();
        userWork.setUserId(userId);

        //获取用户完成工单数
        var workCount = this.getCountByUserId(userId,VMSystem.TASK_STATUS_FINISH,start.toLocalDate(),end);
        userWork.setWorkCount(workCount);
        //获取工单总数
        var total = this.getCountByUserId(userId,null,start.toLocalDate(),end);
        userWork.setTotal(total);
        //获取用户拒绝工单数
        var cancelCount = this.getCountByUserId(userId,VMSystem.TASK_STATUS_CANCEL,start.toLocalDate(),end);
        userWork.setCancelCount(cancelCount);
        //获取进行中得工单数
        var progressTotal = this.getCountByUserId(userId,VMSystem.TASK_STATUS_PROGRESS,start.toLocalDate(),end);
        userWork.setProgressTotal(progressTotal);


        return userWork;
    }

    //根据工单状态，获取用户指定时间段范围的工单数
    private Integer getCountByUserId(Integer userId,Integer taskStatus,LocalDate start,LocalDateTime end){
        var qw = new LambdaQueryWrapper<TaskEntity>();
        qw
                .ge(TaskEntity::getUpdateTime,start)
                .le(TaskEntity::getUpdateTime,end);
        if(taskStatus != null ){
            qw.eq(TaskEntity::getTaskStatus,taskStatus);
        }
        if(userId != null) {
            qw.eq(TaskEntity::getUserId,userId);
        }
        return this.count(qw);
    }

}
